python dictionary comprehension multiple keys